"""For basic init and call"""
import os

from langchain_community.chat_models import QianfanChatEndpoint
from langchain_core.language_models.chat_models import HumanMessage

os.environ["QIANFAN_AK"] = "ae7eoePCWBS8nBTQv7vbmyxm"
os.environ["QIANFAN_SK"] = "tvxwV0Emkp9sxuU3neDfF77Drhl7TWUa"

chat = QianfanChatEndpoint(streaming=True)
print(chat.invoke("你好，可以给我的小狗取个名字吗？").content)